package com.shujia.scala

import scala.io.Source

object Demo20ScalaWordCount {
  def main(args: Array[String]): Unit = {

    //1、读取文件
    val lines: List[String] = Source.fromFile("scala/data/words.txt").getLines().toList

    //2、将每一行中的多个单词拆分出来
    val words: List[String] = lines.flatMap(line => line.split(","))

    //3、对单词进行分组，将同一个单词分到同一个组中
    val groups: Map[String, List[String]] = words.groupBy(word => word)

    //4、统计单词额数量
    val count: Map[String, Int] = groups.map(kv => {
      //一个单词
      val word: String = kv._1
      //d当前单词构建的集合
      val ws: List[String] = kv._2

      //统计单词的数量
      val count: Int = ws.size

      //返回单词和单词的数量
      (word, count)
    })

    /**
      * foreach:  遍历集合，将集合中的元素一个一个传递给后面的函数，没有返回值
      *
      */

    count.foreach(kv => println(kv))


    /**
      * 简写
      *
      */

    //链式调用
    Source
      .fromFile("scala/data/words.txt") //读取文件
      .getLines() //获取所有行
      .toList //转换成list
      .flatMap(_.split(",")) //将单词拆分出来
      .groupBy(word => word) //按照单词进行分组
      .map(kv => (kv._1, kv._2.size)) //统计单词额数量
      .foreach(kv => println(kv)) //打印结果


  }

}
